AIcomes 发表于 2019-4-2 11:50:46

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的版本

Happy清子 发表于 2019-4-3 13:22:55

安装cuda8,是可选项。

Happy清子 发表于 2019-5-15 16:59:44

本教程以安装cuda8 pytorch0.3为例

Happy清子 发表于 2019-5-15 16:59:54

若安装其他的,
参照进行
页: [1]
查看完整版本: conda_DL安装自定义环境(cuda8+pytorch0.3)