百度飞浆AI Studio搭建pytorch框架

1. 检查python版本、GPU版本，查找对应python版本的torch版本和cuda版本，请使用GPU版本启动项目
   
    
2. 打开终端，创建虚拟环境：

   python3 -m venv env_name # (python3 -m 路径 环境名)

   ，项目python版本在创建时就已固定，无法修改，这时要注意torch版本，新建终端用命令nvidia-smi查看cuda版本，如果为12.0则在pytorch的cuda11版本，再检查python版本，如果为3.7则将pip代码最后的118修改为117或116，具体对照表如下：
   
3. 激活新的虚拟环境(linux终端)：

   source env_name/bin/activate

   
   在新的虚拟环境安装对应的GPU版本pytorch

终端命令：

python3.7版本：

​
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

python3.8及以后：

​
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

1. 新开一个终端：输入python：import torch，输入torch.__version__检查torch版本是否正确安装：
   正确安装后，进入项目，切换icoding模式：

切换内核为新建的虚拟环境（如已成功创建新环境但无内核请重启项目）：