服务器配置
ubuntu18.04
anaconda3
cuda10.0
创建虚拟环境
创建名为torch14的虚拟环境,指定python版本为3.6
conda create -n torch14 python=3.6
激活环境
(base) wjw@Super-Server:~/projects$ conda activate torch14
(torch14) wjw@Super-Server:~/projects$ python --version
Python 3.6.12 :: Anaconda, Inc.
安装pytorch1.4
使用官网提供的下载指令,pip会比conda快
(注意服务器CUDA版本,如果是cuda10.0的话最高用pytorch1.4,pytorch1.6要求cuda10.1)