欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

服务器上安装pytorch-gpu

程序员文章站 2022-04-16 08:21:10
1.创建虚拟环境conda create -n pytorch-gpu python=3.6如果需要查看当前有哪些虚拟环境conda info --env2.进入虚拟环境如果启用创建的环境,输入下面指令source activate pytorch-gpu# orconda activate pytorch-gpu退出当前虚拟环境的话,输入指令conda deactivate3.安装pytorch首先要查看服务器的cuda版本,使用如下命令如上图....

1.创建虚拟环境

conda create -n pytorch-gpu python=3.6

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

conda info --env

2.进入虚拟环境

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

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

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

conda deactivate

3.安装pytorch

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

服务器上安装pytorch-gpu

如上图,cuda版本是10.1

或者输入命令

cat /usr/local/cuda/version.txt 

 服务器上安装pytorch-gpu

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

服务器上安装pytorch-gpu

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

conda install pytorch torchvision cudatoolkit=10.0 -c pytorch

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

# 添加Anaconda的清华镜像
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/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes  

然后在输入:

conda install pytorch torchvision cudatoolkit=10.0

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

然后等待下载安装即可。

4.测试

服务器上安装pytorch-gpu

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

参考博客

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

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

 

本文地址:https://blog.csdn.net/sarracode/article/details/107312975