上篇介绍了如何安装Ubuntu安装NVIDIA驱动、CUDA CUDNN 环境,连接如下
https://blog.csdn.net/sinat_42239797/article/details/101618638
今天介绍如何安装pytorch
最好的安装方式是创建一个虚拟环境来安装,因为创建虚拟环境安装不成功或者将来不用了卸载的话比较方便,可以快速卸载并且卸载干净,这样对其他软件没有影响
首先创建一个虚拟环境
conda create -n your_env_name python=X.X(2.7、3.6等) #your_env_name为你的虚拟环境的名字
eg:conda create -n pytorch python=3.7 #方便我创建虚拟环境名字为安装文件名字
等虚拟环境创即按好后要选择激活虚拟环境
conda activate pytorch
然后安装pytorch,我是在官方网站根据自己的环境选择的,官网链接https://pytorch.org
根据给的安装提示在终端输入安装
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
安装完成后进行测试是否安装成功
python
impor