在linux服务器上创建虚拟环境并安装torch

创建环境:

conda create -n your_env_name python=X.X

打开环境:

conda activate your_env_name

安装torch

1、使用nvidia-smi查看服务器cuda版本

2、在pytorch官网查看对应cuda版本的torch安装语句

3、增加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes

4、在命令行输入torch安装语句,但是不要后面的-c pytorch。

然后就可以安装啦!

你可能感兴趣的:(tips,linux,python)