近期有需要用到pytorch的环境,所以就去安装了相应的CUAD和pytorch,现在做一个简单的流程介绍
1.CUDA的安装
对于这个的安装,我主要是借鉴https://zhouchen.blog.csdn.net/article/details/107716721?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-3.control&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromBaidu-3.control这篇博客,可以根据上面的步骤一步一步来,本人安装的是CUDA10.2版本,但我到最后一步
并没有出现如上图所示的画面,而是一闪而过,所以我就去 cmd中输入了
nvcc -V
出现了CUDA相应的版本号,应该也算是安装成功了。
2.安装pytorch
先去anaconda3中创建一个pytorch的虚拟环境(anaconda3的安装步骤这里不多赘述),有两种方法可以进行虚拟环境的创建,
1.直接进入anaconda3中,点击Environments,可以看见你现有的虚拟环境,然后点击左下角的create按钮,就可以进行虚拟环境的创建,如图所示,然后填入环境name进行创建。
2.也可以通过anaconda prompt中输入“conda create -n pytorch36 python=3.6”进行创建,该命令中pytorch36就是虚拟环境的name,python的版本是3.6.
虚拟环境创建完毕,然后进入pytorch官网,https://pytorch.org.找到自己所需的pytorch版本。
tips:需要找到自己所安装的CUDA相应的版本的pytorch。
点击Get Started,可以看见最新的版本号,然后点击Previous PyTorch Versions可以查看历史版本。
复制所选中的那句话,然后进入anaconda prompt中,首先激活你之前所创建的虚拟环境,
conda activate xxxxxx
xxxxx对应的是你虚拟环境的name,成功的话是会将前面括号里面的(base)改成(环境name)
然后将复制的语句粘贴进去,正常情况下会进行安装pytorch,但是我自己安装的时候出现了问题,PackagesNotFoundError: The following packages are not available from current channels
然后发现在该情况下输入
conda config --add channels conda-forge
该指令,然后粘贴之前的pytorch安装命令就可以正常安装pytorch
Proceed 选择“y”,然后就进行下载以及安装,上图是在下载相应的包。安装完毕后,在后面输入python
出现版本号,然后分别输入
import torch
torch.cuda.is_available()
出现true就说明 机器的GPU就可以被Pytorch使用。
也就可以使用该pytorch了!!!