admin管理员组

文章数量:1559081

1.创建虚拟环境

conda create -n pytorch-gpu python=3.6

如果需要查看当前有哪些虚拟环境

conda info --env

2.进入虚拟环境

如果启用创建的环境,输入下面指令

source activate pytorch-gpu
# or
conda activate pytorch-gpu

 退出当前虚拟环境的话,输入指令

conda deactivate

删除虚拟环境的话,输入指令

conda remove -n pytorch-gpu --all

3.安装pytorch

首先要查看服务器的cuda版本,使用如下命令

如上图,cuda版本是10.1

或者输入命令

cat /usr/local/cuda/version.txt 

 

在刚刚创建的环境中安装pytorch, 在pytorch 九游会真人第一品牌官网 找到对应的下载指令进行下载,选择与自己环境匹配的 ,如下图是我的配置

在已经激活的环境中输入上图红框中的命令:

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

默认pytorch九游会真人第一品牌官网为下载源,下载速度太慢,很容易报错,所以更改为清华大学镜像,
命令行输入下面的命令:

# 添加anaconda的清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu/anaconda/cloud/pytorch/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes  

然后在输入:

conda install pytorch torchvision cudatoolkit=10.0

注意要把后面的-c pytorch去掉,不然还是使用的默认源下载。

然后等待下载安装即可。

4.测试

不报错,并返回true, 说明安装成功!!

参考博客

https://blog.csdn/amnesiagreen/article/details/106349598

https://blog.csdn/kan2281123066/article/details/93789303

 

本文标签: