主体

这里记录下我的torch安装过程，本来是使用virtualenv进行安装的，安装是安装成功了，但是导入torchvision的时候出现了问题。

看来以后还是要用conda来进行环境管理比较方便呐！！

进入正题！！！

刚开始我是用virtualenv。出现错误，然后转接用conda试试。

首先安装mini conda。我的系统是ubuntu所以直接wget安装，进入官网直接复制连接就可以了。

下载完后，文件直接就是sh文件，就是脚本文件，直接使用bash 命令，然后就进行安装了。

安装好后，系统提示我没发现conda命令。

下面参考了文章 https://blog.csdn.net/qq_36652619/article/details/79676742

修改环境变量

vim ~/.bashrc

在最后添加：

export PATH=~/anaconda3/bin:$PATH

重启环境变量：

source ~/.bashrc

然后就可以正常使用conda命令了

使用conda创建新的环境，然后在这个环境中安装模块

但还会出现http error。于是csdn后。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda install pytorch torchvision cudatoolkit=9.0

然后就安装成功了

真的很奇怪啊，安装是安装成功了，但是好像不能使用cuda。

torch.version.cuda 如果输出是None的话，则说明不能使用cuda。

怎么解决这个问题呢？

要查看cuda版本是不是和下载的torch版本对应。

nvidia -smi

这里会显示显卡型号，和cuda的版本号

然后去官网下载对应的版本号就行。注意下载的时候把-c去掉，会限制你的下载源

参考文章

详细介绍了 ubuntu系统安装安装torch的过程

https://blog.csdn.net/yimingsilence/article/details/79631567

https://blog.csdn.net/songchunxiao1991/article/details/95192063