安装这个真的踩了很多坑,整天你就光安装吧。
特此记录安装的步骤,防止以后花时间。
首先任务管理器,先确定你的NVIDIA的GPU版本
我的显卡是最差的一个,建议你们买电脑好好看看显卡。
下面查看电脑支持的cuda版本,命令如下:
nvidia-smi
我的cuda版本是10.2,安装pytorch的时候,选择带GPU版本的。
下面安装anaconda3,镜像源:Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror
至少要安装一个conda3版本的,我这里随便选了一个,安装路径选择全英文!
安装好之后,打开conda的命令行:【开始】【所有应用】找到【anaconda】,然后打开:
下面是一些conda操作:
查看所有的虚拟环境:
conda env list
创建虚拟环境
conda create -n pytorch python=3.8
进入pytorch虚拟环境
conda activate pytorch
退出当前环境到base
conda deactivate
查看当前环境安装的库
pip list
下面在新创建的pytorch环境中,安装pytorch。
打开pytorch官网:PyTorch
点积箭头指的,选择以前的版本。
这里要确定的是torch的版本和cuda版本兼容,Python版本和torch版本兼容!
因为我的cuda是10.2,所以选择带GPU版本的torch的时候,只能向下兼容,这里我选择的是cuda10.2对应的GPU版本的pytorch1.9.1,大家按需选择。还需要确定你上边创建的虚拟环境python和当前pytorch版本是兼容的:比如我的python3.8就是兼容pytorch 1.9.1的。
进入你创建的虚拟环境pytorch,然后粘贴上进行运行安装,这里的-c pytorch 是从官网下载可能比较慢。-c是通道的意思。
conda install pytorch==1.9.1 torchvision==0.10.1 torchaudio==0.9.1 cudatoolkit=10.2 -c pytorch
可以选择使用清华源镜像进行加速,只需要将上边的命令后边的-c pytorch改成:
-c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
等待就可以,如果还是慢,可能是网速的原因,建议使用流量,或者手机热点。
最后成功安装!进入Python解释器,输入相应的命令,查看如果是true就是成功了。
安装过程中踩的坑!!!!
问题一:
解决办法:
.condarc文件的路径在:C:\Users\你的用户名\
将.condarc文件的所有内容改成下边:
channels:
- defaults
show_channel_urls: true
default_channels:
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
- https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
pytorch-lts: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
deepmodeling: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/
问题二:anaconda创建虚拟环境报错!
错误: 然后出现一大堆,还有乱码什么的.这里就不展示了.
解决办法:
关掉vpn,重新打开你的命令行,然后重新执行。