conda_DL安装自定义环境(cuda8+pytorch0.3)
本帖最后由 AIcomes 于 2019-4-3 11:40 编辑1- 打开终端,conda 创建自定义环境(cuda8+pytorch0.3+python3.5)步骤如下:
conda create --prefix=/home/ubuntu/MyFiles/.conda/diy python=3.5
conda activate /home/ubuntu/MyFiles/.conda/diy
conda install pytorch=0.3 torchvision cudatoolkit=8.0 -c pytorch
2-检查cuda版本
在终端输入conda list,可以看出CUDA是8.0的版本
3-查看pytorch的版本
安装cuda8,是可选项。 本教程以安装cuda8 pytorch0.3为例 若安装其他的,
参照进行
页:
[1]