window10+ NVIDIA GeForce RTX 3090安装pytorch_综合

显卡版本是 NVIDIA GeForce RTX 3090

安装NVIDIA cudn 11.1

cudnn是8.0.4.30版本

安装显卡见https://blog.csdn.net/china_xin1/article/details/109824882

创建虚拟环境

conda create -n pytorch python=3.6

激活虚拟环境

conda activate pytorch

进入官网选择合适的版本

https://pytorch.org/

conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch

换国内源

没有更改会下载很慢甚至包下载失败

找到user/目录下的.condarc文件改为

channels:- https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
show_channel_urls: true
ssl_verify: true

或者直接在虚拟环境输入切换中科大环境

conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/menpo/conda config --set show_channel_urls yes

https://blog.csdn.net/weixin_43507693/article/details/109015177