在复现代码的过程中需要用到的环境配置为python==3.5.6 torch==1.0.0 CUDA==9.0
pytorch官网中给出了两种安装方式:
- conda
# CUDA 10.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
# CUDA 9.0
conda install pytorch==1.0.0 torchvision==
在复现代码的过程中需要用到的环境配置为python==3.5.6 torch==1.0.0 CUDA==9.0
pytorch官网中给出了两种安装方式:
# CUDA 10.0
conda install pytorch==1.0.0 torchvision==0.2.1 cuda100 -c pytorch
# CUDA 9.0
conda install pytorch==1.0.0 torchvision==