conda_DL安装自定义环境（cuda8+pytorch0.3）

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

本帖最后由 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的版本
torch0.3.jpg

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

若安装其他的，
参照进行

[开发环境] conda_DL安装自定义环境（cuda8+pytorch0.3）