1、配置python环境,python3.8
2、安装Anaconda最新版(建议通过清华镜像下载安装包)
3、配置CUDA11.4(前提是电脑有NVIDIA显卡)
(1)网上好多推荐使用Conda,先添加清华镜像路径
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/# 设置搜索时显示通道地址conda config --set show_channel_urls yes可以在C盘用户目录下找到.condarc的文件,内容如下所示,如果带有—defualts务必删掉,还要确保都是http而不是https(不太懂为什么,不过改过来就可以正常下载)
channels: - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/show_channel_urls: trueremote_read_timeout_secs: 600.0进入pytorch官网,选择合适的安装命令
以最新版本为例,管理员身份进入cmd,输入如下命令(因为使用清华镜像下载,所以不要带-c pytorch)
(2)使用pip下载安装pytorch
进入pytorch官网,选择合适的安装命令
以最新版本为例,管理员身份进入cmd,输入如下命令
后面加的-i https://pypi.tuna.tsinghua.edu.cn/simple就表示通过清华镜像下载。
测试安装结果:在cmd中输入
python进入python编译器,输入
import torchtorch.__version__输出为相应的版本号:
'1.9.0+cu111'