1、安装pytorch需要提前安装好anaconda 和 cuda
cuda安装教程:nvcc -V不是内部或外部命令,也不是可运行的程序_Nefu_lyh的博客-CSDN博客
2.打开Anaconda Prompt; 更换镜像源,创建虚拟环境;
配置anaconda镜像源:(清华镜像源)-------下载加速用;
conda config --add channels Index of /anaconda/pkgs/free/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
conda config --add channels Index of /anaconda/pkgs/main/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
conda config --add channels Index of /anaconda/cloud/pytorch/win-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
conda config --set show_channel_urls yes
查看当前使用源:
conda config --show-sources
创建虚拟环境命令:conda create -n pytorch python=3.6 (python版本看自己情况,最好旧一点)
注意,这里的 pytorch 是虚拟环境的名称,可随意取,最好取pytorch
创建好后会提示
3、进入pytorch虚拟环境:
命令:conda activate pytorch
查看安装的库:
4、安装pytorch并验证;
4.1 先查看当前计算机的cuda版本(英伟达版本):nvidia-smi
若显卡过旧:如1050显卡过旧;在英伟达控制面板-帮助-系统信息-组件-cuda版本中查看;
4.2 选择对应的pytorch版本安装 (我:1050显卡,cuda版本9.2,安装的1.2版本的pytorch)
5.安装完成;进入python,调用torch库,torch.cuda.is_available()函数,返回true则完成
>