window10+ NVIDIA GeForce RTX 3090安装pytorch

最新推荐文章于 2024-02-13 11:26:26 发布

sereasuesue

最新推荐文章于 2024-02-13 11:26:26 发布

阅读量2.6k

点赞数 1

分类专栏：问题记录环境搭建等文章标签： anaconda cuda pytorch安装

本文链接：https://blog.csdn.net/sereasuesue/article/details/112326433

版权

问题记录环境搭建等专栏收录该内容

35 篇文章 5 订阅

订阅专栏

显卡版本是 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