创建环境:
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。
然后就可以安装啦!